#pragma once

#include <string>
#include <unistd.h>
#include <iostream>
#include <functional>
#include <pthread.h>


//Version 2
// namespace ThreadModule
// {
//     enum class TSTATUS
//     {
//         NEW,
//         RUNNING,
//         STOP
//     };
//     static int number = 0;

//     template<class T>
//     class Thread
//     {
//         using threadfunc_t = std::function<void(T&)>;
//         static void *routine(void *args)
//         {
//             Thread* t = static_cast<Thread*>(args);
//             t->_status = TSTATUS::RUNNING;
//             t->_func(t->_data);
//             return nullptr;
//         }
//     public:

//         Thread(threadfunc_t func, T& data) : _func(func), _isjoined(true),
//         _data(data)
//         {
//             _name = "thread-" + std::to_string(number++);
//             _status = TSTATUS::NEW;
//         }

//         bool Start()
//         {
//             if (_status == TSTATUS::NEW)
//             {
//                 int n = pthread_create(&_id, nullptr, routine, this);
//                 if (n != 0)
//                 {
//                     perror("pthread_create error");
//                     return false;
//                 }
//                 return true;
//             }
//             return false;
//         }
//         bool Join()
//         {
//             if (_isjoined)
//             {
//                 int n = pthread_join(_id, nullptr);
//                 if (n != 0)
//                 {
//                     perror("pthread_join error");
//                     return false;
//                 }
//                 return true;
//             }
//             return false;
//         }
//         bool Stop()
//         {
//             if (_status == TSTATUS::RUNNING)
//             {
//                 int n = pthread_cancel(_id);
//                 if (n != 0)
//                 {
//                     perror("pthread_cancel error");
//                     return false;
//                 }
//                 _status = TSTATUS::STOP;
//                 return true;
//             }
//             return false;
//         }
//         void EnableDetach() {
//             _isjoined  = false;
//         }
//         bool IsJoinable() {
//             return _isjoined;
//         }
//         std::string Name() {return _name;}
//         ~Thread() {}

//     private:
//         std::string _name;
//         pthread_t _id;
//         bool _isjoined;
//         TSTATUS _status;
//         threadfunc_t _func;
//         T& _data;
//     };
// }

//Vsersion 1
namespace ThreadModule
{

    using threadfunc_t = std::function<void()>;

    enum class TSTATUS
    {
        NEW,
        RUNNING,
        STOP
    };
    static int number = 0;

    class Thread
    {
        static void *routine(void *args)
        {
            Thread* t = static_cast<Thread*>(args);
            t->_status = TSTATUS::RUNNING;
            t->_func();
            return nullptr;
        }
    public:
        Thread(threadfunc_t func) : _func(func), _isjoined(true)
        {
            _name = "thread-" + std::to_string(number++);
            _status = TSTATUS::NEW;
        }
        std::string Name() {return _name;}
        bool Start()
        {
            if (_status == TSTATUS::NEW)
            {
                int n = pthread_create(&_id, nullptr, routine, this);
                if (n != 0)
                {
                    perror("pthread_create error");
                    return false;
                }
                return true;
            }
            return false;
        }
        bool Join()
        {
            //直接等待就可以
            if (_isjoined)
            {
                int n = pthread_join(_id, nullptr);
                if (n != 0)
                {
                    perror("pthread_join error");
                    return false;
                }
                return true;
            }
            return false;
        }
        bool Stop()
        {
            if (_status == TSTATUS::RUNNING)
            {
                int n = pthread_cancel(_id);
                if (n != 0)
                {
                    perror("pthread_cancel error");
                    return false;
                }
                _status = TSTATUS::STOP;
                return true;
            }
            return false;
        }
        void EnableDetach() {
            _isjoined  = false;
        }
        bool IsJoinable() {
            return _isjoined;
        }

        ~Thread() {}

    private:
        std::string _name;
        pthread_t _id;
        bool _isjoined;
        TSTATUS _status;
        threadfunc_t _func;
    };
}